# myapp/settings.py

MIDDLEWARE = [
    # 'opencensus.ext.django.middleware.OpencensusMiddleware',
    'myapp.middleware.PatchedOpencensusMiddleware',
]

# myapp/middleware.py

import django
from opencensus.ext.django.middleware import (
    OpencensusMiddleware,
    EXCLUDELIST_PATHS,
    EXCLUDELIST_HOSTNAMES,
    _trace_db_call
)
import six
from django.db import connection
from opencensus.common import configuration
from opencensus.trace import (
    integrations,
    print_exporter,
    samplers,
)
from opencensus.trace.propagation import trace_context_http_header_format


class PatchedOpencensusMiddleware(OpencensusMiddleware):
    def __init__(self, get_response=None):
        super(OpencensusMiddleware, self).__init__(get_response)
        settings = getattr(django.conf.settings, 'OPENCENSUS', {})
        settings = settings.get('TRACE', {})

        self.sampler = (settings.get('SAMPLER', None)
                        or samplers.ProbabilitySampler())
        if isinstance(self.sampler, six.string_types):
            self.sampler = configuration.load(self.sampler)

        self.exporter = settings.get('EXPORTER', None) or \
            print_exporter.PrintExporter()
        if isinstance(self.exporter, six.string_types):
            self.exporter = configuration.load(self.exporter)

        self.propagator = settings.get('PROPAGATOR', None) or \
            trace_context_http_header_format.TraceContextPropagator()
        if isinstance(self.propagator, six.string_types):
            self.propagator = configuration.load(self.propagator)

        self.excludelist_paths = settings.get(EXCLUDELIST_PATHS, None)

        self.excludelist_hostnames = settings.get(EXCLUDELIST_HOSTNAMES, None)

        if django.VERSION >= (2,):  # pragma: NO COVER
            connection.execute_wrappers.append(_trace_db_call)

        # pylint: disable=protected-access
        integrations.add_integration(integrations._Integrations.DJANGO)